#ifndef YOLOV5_H
#define YOLOV5_H

#include <vector>
#include "utils.h"

class YoloV5
{
public:
    YoloV5();
    ~YoloV5();

    int open(int modelWidth, int modelHeight, int clsNum);
    int init(const unsigned char* modelData, int modelSize);
    int detect(const unsigned char* pData, std::vector<BBOX>& detectResults);
    void getModelRequstSize(int& width, int& height);
private:
    int m_clssNum = 8;
    unsigned long long m_rknnCtx = 0;
    int m_requestImgWidth = 640;
    int m_requestImgHeight = 640;
    int m_requestImgChannel = 3;
    int m_tensorInputNum;
    int m_tensorOutputNum;
};


#endif